package com.zjhd.sealcutting.utils;

import com.zjhd.sealcutting.domain.UserInfo;
import com.zjhd.sealcutting.exception.BizException;
import io.jsonwebtoken.*;
import lombok.SneakyThrows;
import org.apache.commons.lang.StringUtils;
import org.springframework.http.server.reactive.ServerHttpRequest;
import javax.crypto.spec.SecretKeySpec;
import javax.xml.bind.DatatypeConverter;
import java.security.Key;
import java.util.Date;
import org.joda.time.DateTime;
/**
 * TODO  Token工具类
 *
 * @author wangchao
 * @version 1.0
 * @date 2020/11/24 10:00
 */
public class TokenUtil {

    /**
     * 签名秘钥,可以换成 秘钥 注入
     */
    public static final String SECRET = "seal_cutting";//注意:本参数需要长一点，不然后面剪切的时候很可能长度为0，就会报错
    /**
     * 签发地
     */
    public static final String issuer = "com.zjhd.sealcutting";
    /**
     * 过期时间
     */
    public static final int expire = 14400;

    /**
     * 生成token
     *
     * @return
     */
    public static String createJwtToken(String subject, UserInfo user) {
        return createJwtToken( issuer, subject, expire,user);
    }
    public static String createJwtToken(UserInfo user) {
        return createJwtToken( issuer, "", expire,user);
    }

    /**
     * 生成Token
     * @param issuer  该JWT的签发者，是否使用是可选的
     * @param subject  该JWT所面向的用户，是否使用是可选的；
     * @param expire 签发时间 （有效时间，过期会报错）
     * @return token String
     */
    public static String createJwtToken(String issuer, String subject, int expire,UserInfo user) {

        // 签名算法 ，将对token进行签名
        SignatureAlgorithm signatureAlgorithm = SignatureAlgorithm.HS256;
        // 通过秘钥签名JWT
        byte[] apiKeySecretBytes = DatatypeConverter.parseBase64Binary(SECRET);
        String str=signatureAlgorithm.getJcaName();
        Key signingKey = new SecretKeySpec(apiKeySecretBytes, str);
        // 让我们设置JWT声明
        if(null != user){
            JwtBuilder builder = Jwts.builder().setId(String.valueOf(user.getId()))
                    .setSubject(subject)
                    .setIssuer(issuer)
                    .claim(CommonConstants.JWT_KEY_USER_ID, user.getId())
                    .claim(CommonConstants.JWT_KEY_NAME, user.getUserName())
                    .claim(CommonConstants.JWT_KEY_MOBILE, user.getMobile())
                    .setExpiration(DateTime.now().plusSeconds(expire).toDate())
                    .signWith(signatureAlgorithm, signingKey);
            // 构建JWT并将其序列化为一个紧凑的url安全字符串
            return builder.compact();
        }else{
            return null ;
        }
    }

    /**
     * Token解析方法
     * @param jwt Token
     * @return
     */
    public static Claims parseJWT(String jwt) {
        Claims claims;
        try {
            claims  = Jwts.parser()
                    .setSigningKey(DatatypeConverter.parseBase64Binary(SECRET))
                    .parseClaimsJws(jwt).getBody();
        }catch (ExpiredJwtException e){
            e.printStackTrace();
            // 如果这行代码不是签名的JWS(如预期)，那么它将抛出异常
            claims=e.getClaims();
        }
        return claims;
    }

    @SneakyThrows
    public static void checkTokenIsLev(ServerHttpRequest request, String token){
        try {
            Claims claims = parseJWT(token) ;
            Date expiration = claims.getExpiration();
            //和当前时间进行对比来判断是否过期
            boolean bool=new Date(System.currentTimeMillis()).after(expiration);
            if(bool){
                throw new BizException("4005","您已离线，请重新登录");
            }
        }catch (Exception e){
            e.printStackTrace();
            throw new BizException("4005","Token异常，请重新登录");
        }
    }

    /**
     * TODO  获取token中用户信息
     * @author wangchao
     * @date 2020/11/24 10:58
     * @param
     * @return
     */
    public static UserInfo parseAccessToken(String token) throws Exception {
        try {
            Claims body = parseJWT(token);
            UserInfo user=new UserInfo();
            if(body!=null) {
                String id = StringHelper.getObjectValue(body.get(CommonConstants.JWT_KEY_USER_ID));
                if (StringUtils.isNotBlank(id)) {
                    user.setId(Integer.parseInt(id));
                }
                user.setUserName(StringHelper.getObjectValue(body.get(CommonConstants.JWT_KEY_NAME)));
                user.setMobile(StringHelper.getObjectValue(body.get(CommonConstants.JWT_KEY_MOBILE)));
                user.setUserGroup(StringHelper.getObjectValue(body.get(CommonConstants.JWT_KEY_MOBILE)));
            }
            return user;
        }catch (Exception e){
            e.printStackTrace();
        }
        return null;
    }
}
